"""
    主界面
"""
from time import sleep

from selenium import webdriver
from selenium.webdriver.common.by import By

from src.web.wework.contacts_page import ContactsPage


class MainPage:

    def __init__(self, driver):
        self.driver = driver

    def go_menu_contacts(self):
        """点击通讯录"""
        self.driver.find_element(By.CSS_SELECTOR, "#menu_contacts").click()
        sleep(1)
        return ContactsPage(self.driver)
